Skip to content

New active learning with AiiDA and FLARE#414

Open
bastonero wants to merge 27 commits into
SSCHAcode:masterfrom
bastonero:new/flare-interface
Open

New active learning with AiiDA and FLARE#414
bastonero wants to merge 27 commits into
SSCHAcode:masterfrom
bastonero:new/flare-interface

Conversation

@bastonero
Copy link
Copy Markdown
Contributor

We here implement a new active learning protocol using the FLARE machine-learning interatomic potential. FLARE provides prediction uncertainties that can be used to actively learn the potential energy surface of the material.

The description of this new scheme will be made available soon through an arXiv publication. "Lorenzo Bastonero, Gabriel Joalland, Chiara Cignarella, Lorenzo Monacelli, and Nicola Marzari, Predicting challenging phase transitions in energy materials with Bayesian active learning".

Examples are provided within the dedicated folder.

bastonero and others added 26 commits September 8, 2023 12:57
I implement in the `AiiDAEnsemble` a first version of
on-the-fly training using the FLARE code, which exploits
(sparse) Gaussian processes and the ACE/MB descriptors
to train and make estimations of the errors.
Examples for running SSCHA using AiiDA+FLARE
to perform the SSCHA minimization are added.
These examples serve as well as for testing some
hard-to-test logic within the OTF workflow.
The mir-flare python package must be installed to perform
the tests during the continous integration workflow.
In AiiDAEnsemble it is added the possibility of splitting the
submission of all the calculations in a certain number of batches.
This is convenient especially when performing the calculations
using active-learning/on-the-fly simulations.
The `self.init()` wasn't called, and it was messing up the minimization.
A variable controlling when the training is performed is added.
This can give improvements if the hyperparameter training gets
stucked into local minima due to few environments in the dataset.
@mesonepigreco mesonepigreco self-requested a review April 18, 2026 19:01
Comment thread Examples/sscha_and_aiida/debug.ipynb Outdated
Comment thread Examples/sscha_and_aiida/write_xyz.ipynb Outdated
Comment thread .pre-commit-config.yaml Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants